package com.ujs.listnode.code.RemoveNode;

import com.ujs.listnode.code.ListNode;

/**
 * 双向链表
 *
 * @author zhangshihao
 * @create 2023-09-26 16:21
 */
public class MyDoubleLinkedList {
    int size = 0;
    ListNode head, tail;

    public MyDoubleLinkedList() {
        // 初始化操作
        this.size = 0;
        this.head = new ListNode(0);
        this.tail = new ListNode(0);
        // 这一步非常关键，否则在加入头结点的操作中会出现null.next的错误！
        head.next = tail;
        tail.pre = head;
    }

    public void addAtIndex(int index, int val) {
        if (index > size) {
            return;
        }
        if (index < 0) {
            index = 0;
        }
        size++;
        // 找到插入位置的前一个节点
        ListNode pre = head;
        for (int i = 0; i < index; i++) {
            pre = pre.next;
        }
        ListNode newNode = new ListNode(val);
        newNode.next = pre.next;
        pre.next.pre = newNode;
        newNode.pre = pre;
        pre.next = newNode;
    }

    public int get(int index) {
        // 判断index是否有效
        if (index < 0 || index >= size) {
            return -1;
        }
        ListNode cur = head;
        // 判断哪一边遍历的时间更短
        if (index >= size / 2) {
            // 从后往前遍历
            cur = tail;
            for (int i = 0; i < size - index; i++) {
                cur = cur.pre;
            }
        } else {
            // 从前往后遍历
            for (int i = 0; i < index; i++) {
                cur = cur.next;
            }
        }
        return cur.val;
    }

    public void addAtHead(int val) {
        // 等价于在第0个元素前添加
        addAtIndex(0, val);
    }

    public void addAtTail(int val) {
        // 等价于在最后一个元素(null)前添加
        addAtIndex(size, val);
    }

    public void deleteAtIndex(int index) {
        if (index >= size || index < 0) {
            return;
        }
        // 找到前驱
        ListNode pre = head;
        for (int i = 0; i < index; i++) {
            pre = pre.next;
        }
        pre.next = pre.next.next;
        pre.next.next.pre = pre;
    }
}
